package leetcode.题库.数组.普通数组;

/**
 * 两数之和
 *
 * @author yanzhenming
 * @date 2023/3/22  15:02
 */

import java.util.HashMap;
import java.util.Map;

/**
 * 描述
 *
 * 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值 target  的那两个整数，并返回它们的数组下标。
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
 * 你可以按任意顺序返回答案。
 *
 */

/**
 * 思路
 * 时间换空间，用map将时间复杂度降低为O(n)
 */
public class Solution1 {
    public static void main(String[] args) {
        int[] arr = new int[]{2,7,11,15};
        System.out.println(twoSum(arr, 9)[0]);
        System.out.println(twoSum(arr, 9)[1]);
    }

    public static int[] twoSum(int[] nums, int target) {
        int[] res = new int[2];
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int subNum = target - nums[i];
            if (map.containsKey(subNum)) {
                res[0] = map.get(subNum);
                res[1] = i;
                break;
            }
            map.put(nums[i], i);
        }
        return res;
    }
}
